/***************************************************************************
 *   Copyright (C) 2008 by BogDan Vatra                                    *
 *   bogdan@licentia.eu                                                    *
 *                                                                         *
 *   This program is free software: you can redistribute it and/or modify  *
 *   it under the terms of the GNU General Public License as published by  *
 *   the Free Software Foundation, either version 3 of the License, or     *
 *   (at your option) any later version.                                   *
 *   This program is distributed in the hope that it will be useful,       *
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
 *   GNU General Public License for more details.                          *
 *   You should have received a copy of the GNU General Public License     *
 *   along with this program.  If not, see <http://www.gnu.org/licenses/>. *
 ***************************************************************************/
#ifndef SQLQUERYMODEL_H
#define SQLQUERYMODEL_H

#include <QAbstractTableModel>
#include <QStringList>
#include <QVector>
#include "win32def.h"
#include "tableview.h"

namespace SQL
{
class Query;

class  QueryModel : public QAbstractTableModel
{
	Q_OBJECT
	QueryModel(Query *parent, const QStringList & columns = QStringList(), const QStringList & columnNames = QStringList());
	friend class Query;

public:
	~QueryModel();
	int rowCount(const QModelIndex & parent = QModelIndex()) const;
	int columnCount(const QModelIndex & parent = QModelIndex()) const;
	void setColumns(const QStringList & columns);
	void setColumnNames(const QStringList & columnNames);
	QVariant headerData(int section, Qt::Orientation orientation, int role = Qt::DisplayRole) const;

	void setCheckableColumn(const QString& checkableColumn);
	void checkAll();
	void uncheckAll();
	const QVector<bool> & checkedRows();
	QVariant data(const QModelIndex & index, int role = Qt::DisplayRole) const;
	bool setData ( const QModelIndex & index, const QVariant & value, int role = Qt::EditRole );
	Qt::ItemFlags flags(const QModelIndex & index) const;
        void setView(TableView*);

        void sort ( int column, Qt::SortOrder order = Qt::AscendingOrder );

private:
	Query *m_parent;
	QStringList m_columns;
	QStringList m_columnNames;
	mutable int m_lastRow;
	QString m_checkableColumn;
	QVector<bool> m_checkedRows;
        TableView *m_view;
public slots:
	void afterSelect();
};

}

#endif
